package com.algorithm.liyc.practice;

/**
 * 实现 strStr() 函数。
 * 给定一个 haystack 字符串和一个 needle 字符串，在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在，则返回  -1。
 * KMP的经典思想就是:当出现字符串不匹配时，可以记录一部分之前已经匹配的文本内容，利用这些信息避免从头再去做匹配。
 *
 * @author Liyc
 * @date 2023/12/14 15:10
 **/

public class Solution28 {
    public int strStr(String haystack, String needle) {
        if (needle.length() == 0){
            return -1;
        }
        int[] next = new int[needle.length()];
        getNext(next, needle);
        int j = 0;
        for (int i = 0; i < haystack.length(); i++) {
            while (j > 0 && haystack.charAt(i) != needle.charAt(j)) {
                j = next[j-1];
            }
            if (haystack.charAt(i) == needle.charAt(j)) {
                j++;
            }
            if (j == needle.length()) {
                return i - needle.length() + 1;
            }
        }
        return -1;
    }

    /**
     * 获取前缀表
     * 构造next数组其实就是计算模式串s，前缀表的过程。 主要有如下三步：
     * 1. 初始化
     * 2. 处理前后缀不相同的情况
     * 3. 处理前后缀相同的情况
     * @param next
     * @param s
     */
    public void getNext(int[] next, String s) {
        int j = 0;
        next[0] = j;
        for (int i = 1; i < s.length(); i++) {
            while (j > 0 && s.charAt(i) != s.charAt(j + 1)) {
                j = next[j - 1];
            }
            if (s.charAt(i) == s.charAt(j + 1)) {
                j++;
            }
            next[i] = j;
        }
    }
}
